package config

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

var Users = map[string][]string{
	"ftp":        {"ftp", "admin", "www", "web", "root", "db", "wwwroot", "data", "sys", "system", "ftptest"},
	"mysql":      {"root", "mysql", "admin"},
	"mssql":      {"sa", "sql"},
	"smb":        {"administrator", "admin", "guest"},
	"rdp":        {"administrator", "admin", "guest"},
	"postgresql": {"postgres", "admin"},
	"ssh":        {"root", "admin", "system", "test"},
	"mongodb":    {"root", "admin"},
	"winrm":      {"admin", "administrator", "test"},
	"redis":      {"default", "system", "test", "admin", "root"},
	"BaseAuth":   {"administrator", "admin", "guest", "system", "root", "sys"},
}

var Passwords = []string{"12345", "123123", "123456", "admin", "admin123", "root", "", "test123456", "65535", "pass123", "pass@123", "password", "123123", "654321", "111111", "123", "1", "admin@123", "Admin@123", "admin123!@#", "system1", "system111", "system@123", "system_123", "system#123", "system@111", "system@2019", "system@2020", "system@2021", "system@2022", "system@2023", "system@123#4", "admin1", "admin111", "admin@123", "admin_123", "admin#123", "admin@111", "admin@2019", "admin@2020", "admin@2021", "admin@2022", "admin@2023", "admin@123#4", "root1", "root111", "root@123", "root_123", "root#123", "root@111", "root@2019", "root@2020", "root@2021", "root@2022", "root@2023", "root@123#4", "test1", "test111", "test@123", "test_123", "test#123", "test@111", "test@2019", "test@2020", "test@2021", "test@2022", "test@2023", "test@123#4", "P@ssw0rd!", "P@ssw0rd", "Passw0rd", "qwe123", "12345678", "test", "test123", "123qwe", "123qwe!@#", "123456789", "123321", "666666", "a123456.", "123456~a", "123456!a", "000000", "1234567890", "8888888", "!QAZ2wsx", "1qaz2wsx", "abc123", "abc123456", "1qaz@WSX", "a11111", "a12345", "Aa1234", "Aa1234.", "Aa12345", "a123456", "a123123", "Aa123123", "Aa123456", "Aa12345.", "sysadmin", "system", "1qaz!QAZ", "2wsx@WSX", "qwe123!@#", "Aa123456!", "A123456s!", "sa123456", "1q2w3e", "Charge123", "Aa123456789"}

var DefaultPorts = "7,21,22,23,25,43,49,53,69,70,79,80,81,82,83,84,85,86,87,88,89,90,99,100,109,110,111,113,119,125,135,139,143,146,161,162,163,194,199,254,255,256,259,264,389,443,444,445,458,464,465,481,497,500,512,513,514,515,541,543,544,545,548,554,555,563,587,593,616,617,625,631,636,646,648,666,667,668,683,687,691,800,801,808,843,873,880,888,898,900,901,902,903,911,912,981,987,990,992,993,995,999,1000,1001,1002,1007,1009,1010,1011,1021,1022,1023,1024,1025,1026,1027,1028,1029,1030,1031,1032,1033,1034,1035,1036,1037,1038,1039,1040,1041,1042,1043,1044,1045,1046,1047,1048,1049,1050,1051,1052,1053,1054,1055,1056,1057,1058,1059,1060,1061,1062,1063,1064,1065,1066,1067,1068,1069,1070,1071,1072,1073,1074,1075,1076,1077,1078,1079,1080,1081,1082,1083,1084,1085,1086,1087,1088,1089,1090,1091,1092,1093,1094,1095,1096,1097,1098,1099,1100,1102,1104,1105,1106,1107,1108,1110,1111,1112,1113,1114,1117,1119,1121,1122,1123,1124,1126,1130,1131,1132,1137,1138,1141,1145,1147,1148,1149,1151,1152,1154,1163,1164,1165,1166,1169,1174,1175,1183,1185,1186,1187,1192,1198,1199,1201,1213,1216,1217,1218,1233,1234,1236,1244,1247,1248,1259,1271,1272,1277,1287,1296,1300,1301,1309,1310,1311,1322,1328,1334,1352,1417,1433,1434,1443,1455,1461,1494,1500,1501,1503,1521,1524,1533,1556,1580,1583,1594,1600,1641,1658,1666,1687,1688,1700,1717,1718,1719,1720,1721,1723,1755,1761,1782,1783,1801,1805,1812,1839,1840,1862,1863,1864,1875,1900,1914,1935,1947,1971,1972,1974,1984,1998,1999,2049,2128,2288,2301,2323,2375,2376,2381,3128,3168,3211,3221,3260,3261,3268,3269,3283,3299,3300,3306,3322,3323,3324,3325,3333,3351,3367,3369,3370,3371,3372,3389,3390,3689,3690,3703,3737,3766,3784,3800,3801,3809,3814,3826,3827,3828,3851,3869,3871,3878,3880,3889,3905,3914,3918,3920,3945,3971,3986,3995,3998,4000,4001,4002,4003,4004,4005,4006,4045,4111,4125,4126,4129,4224,4242,4279,4321,4343,4443,4444,4445,4446,4449,4550,4567,4662,4848,4899,4900,4998,5000,5001,5002,5003,5004,5009,5030,5033,5050,5051,5054,5060,5061,5080,5087,5100,5101,5102,5120,5190,5200,5214,5221,5222,5225,5226,5269,5280,5298,5357,5405,5414,5431,5432,5440,5500,5510,5544,5550,5555,5560,5566,5631,5632,5633,5666,5678,5679,5718,5730,5800,5801,5802,5810,5811,5815,5822,5825,5850,5859,5862,5877,5900,5901,5902,5903,5904,5906,5907,5910,5911,5915,5922,5925,5950,5952,5959,5960,5961,5962,5963,5987,5988,5989,5998,5999,6000,6001,6002,6003,6004,6005,6006,6007,6009,6025,6059,6100,6101,6106,6112,6123,6129,6156,6346,6379,6389,6502,6510,6543,6547,6565,6566,6567,6580,6646,6666,6667,6668,6669,6689,6692,6699,6779,6788,6789,6792,6839,6881,6901,6969,7000,7001,7002,7004,7007,7019,7025,7070,7100,7103,7106,7200,7201,7402,7435,7443,7496,7512,7625,7627,7676,7741,7777,7778,7800,7911,7920,7921,7937,7938,7999,8000,8001,8002,8007,8008,8009,8010,8011,8021,8022,8031,8042,8045,8080,8081,8082,8083,8084,8085,8086,8087,8088,8089,8090,8093,8099,8100,8180,8181,8192,8193,8194,8200,8222,8254,8290,8291,8292,8300,8333,8383,8400,8402,8443,8500,8600,8649,8651,8652,8654,8701,8800,8873,8888,8899,8994,9000,9001,9002,9003,9009,9010,9011,9040,9050,9071,9080,9081,9090,9091,9099,9100,9101,9102,9103,9110,9111,9200,9300,9500,9502,9503,9535,9575,9593,9594,9595,9618,9666,9876,9877,9878,9898,9900,9917,9929,9943,9944,9968,9998,9999,10000,10001,10002,10003,10008,10009,10010,10012,10024,10025,10082,10180,10215,10243,10566,10616,10617,10621,10626,10628,10629,10778,11110,11111,11211,11967,12000,12174,12265,12345,13456,13722,13782,13783,14000,14238,14441,14442,15000,15002,15003,15004,15660,15742,16000,16001,16012,16016,16018,16080,16113,16992,16993,17877,17988,18040,18101,18988,19101,19283,19315,19350,19780,19801,19842,20000,20005,20031,20221,20222,20828,21571,22939,23502,24444,24800,25734,25735,26214,27000,27352,27353,27355,27356,27715,28201,30000,30718,30951,31038,31337,32768,32769,32770,32771,32772,32773,32774,32775,32776,32777,32778,32779,32780,32781,32782,32783,32784,32785,33354,33899,34571,34572,34573,35500,38292,40193,40911,41511,42510,44176,44442,44443,44501,45100,48080,49152,49153,49154,49155,49156,49157,49158,49159,49160,49161,49163,49165,49167,49175,49176,49400,49999,50000,50001,50002,50003,50006,50300,50389,50500,50636,50800,51103,51493,52673,52822,52848,52869,54045,54328,55055,55056,55555,55600,56737,56738,57294,57797,58080,60020,60443,61532,61900,62078,63331,64623,64680,65000,65129,65389,91,92,98,889,1118,1888,2375,2379,3008,3505,6080,6648,6868,7003,7005,7008,7071,7074,7078,7080,7088,7680,7687,7688,7890,8003,8004,8006,8012,8016,8018,8020,8028,8030,8038,8044,8046,8048,8053,8060,8069,8070,8091,8092,8094,8095,8096,8097,8098,8101,8108,8118,8161,8172,8244,8258,8280,8288,8360,8448,8484,8834,8838,8848,8858,8868,8879,8880,8881,8983,8989,9008,9043,9060,9082,9083,9084,9085,9086,9087,9088,9089,9092,9093,9094,9095,9096,9097,9098,9443,9448,9800,9981,9986,9988,10008,10250,12018,12443,18000,18001,18002,18008,18080,18082,18088,18090,18098,18888,19001,20720,21000,21501,21502,28018,20880,27017"

var BasicPorts = "21,22,80,81,135,139,443,445,1433,3306,5432,6379,7001,8000,8080,8088,8089,8090,9000,9200,9300,11211,27017"

func UserIsExist(userDict string) bool {
	if IsExist(userDict) {
		return true
	}
	return false
}

func PassIsExist(passDict string) bool {
	if IsExist(passDict) {
		return true
	}
	return false
}

func IsExist(f string) bool {
	_, err := os.Stat(f)
	return err == nil || os.IsExist(err)
}

func UserDict(userDict string) (user []string) {
	file, err := os.Open(userDict)
	if err != nil {
		fmt.Println("Open "+userDict+" error, %v", err)
	}
	fi, _ := os.Stat(userDict)
	if fi.Size() == 0 {
		fmt.Println("Error: " + userDict + " is null!")
		os.Exit(1)
	}

	defer file.Close()
	scanner := bufio.NewScanner(file)
	scanner.Split(bufio.ScanLines)

	for scanner.Scan() {
		passwd := strings.TrimSpace(scanner.Text())
		if passwd != "" {
			user = append(user, passwd)
		}
	}
	return user
}

func PassDict(passDict string) (pass []string) {
	file, err := os.Open(passDict)
	if err != nil {
		fmt.Println("Open "+passDict+" error, %v", err)
	}
	fi, _ := os.Stat(passDict)
	if fi.Size() == 0 {
		fmt.Println("Error: " + passDict + " is null!")
		os.Exit(1)
	}

	defer file.Close()
	scanner := bufio.NewScanner(file)
	scanner.Split(bufio.ScanLines)

	for scanner.Scan() {
		passwd := strings.TrimSpace(scanner.Text())
		if passwd != "" {
			pass = append(pass, passwd)
		}
	}
	return pass
}
